import ollama

class ITranslate:
    '''翻译接口'''

class Translate:
    def __init__(self,src_lan="中文", dest_lan="英语"):
        self.src_lan = src_lan
        self.dest_lan = dest_lan

    def translate(self, content):
            response = ollama.chat(model="qwen2:7b", messages=[{
                'role': 'user',
                'content': '你是一名翻译专家，请将以下内容从{}翻译成{}，输出翻译结果即可，内容如下：{}'.format(self.src_lan, self.dest_lan, content)
            }])
            #TODO 需要彻底解决生成纯净的翻译
            # 只获取第一句，去掉后面的补充
            txt = response['message']['content']
            if '\n' in txt:
                txt = txt[:txt.index("\n")]
            return txt

if __name__ == '__main__':
    t = Translate()
    response = t.translate("我是一名中国人")
    print(response)